package config

import (
	"CSCredis/global"
	"fmt"
	"github.com/fsnotify/fsnotify"
	"github.com/spf13/viper"
)

/**
 * @Author: hejing@szlabsun.com
 * @Description:
 * @File:  config
 * @Version: 1.0.0
 * @Date: 2021/6/22 14:44
 */

const defaultConfigFile = "config.yaml"

// 读取配置文件的值 并将其写入到全局变量 global.CONFIG 里
func init() {
	v := viper.New()
	//v.AddConfigPath("$GOPATH/src/")
	//v.SetConfigFile(defaultConfigFile)
	v.SetConfigName("config")
	// 设置 public 配置文件类型
	v.SetConfigType("yaml")
	// 设置配置文件存放的目录
	v.AddConfigPath("./")
	err := v.ReadInConfig()
	if err != nil {
		panic(fmt.Errorf("Fatal error config file: %s \n", err))
	}
	v.WatchConfig()
	v.OnConfigChange(func(e fsnotify.Event) {
		fmt.Println("config file changed:", e.Name)
		if err := v.Unmarshal(&global.CONFIG); err != nil {
			fmt.Println(err)
		}
	})
	if err := v.Unmarshal(&global.CONFIG); err != nil {
		fmt.Println(err)
	}
}
